#include <iostream>
#include <vector>
#include <functional>
#include <cstdio>
#include <cstring>
#include <cerrno>
#include <unistd.h>
#include <fcntl.h>

using namespace std;

//准备一批任务，在IO非阻塞式等待时，执行这些任务
using func_t = function<void(void)>;

//打印日志
void PrintLOG()
{
    cout << "正在执行日志打印任务" << endl;
}

//数据库查询
void SelectSQL()
{
    cout << "正在执行数据库查询任务" << endl;
}

//网络检测
void NetCheck()
{
    cout << "正在执行网络检测任务" << endl;
}

//装载任务
vector<func_t> vfcs;

void LoadTask()
{
    vfcs.push_back(PrintLOG);
    vfcs.push_back(SelectSQL);
    vfcs.push_back(NetCheck);
}

//启动任务
void Start()
{
    for(const auto &e : vfcs)
        e();
}

void setNonBlock(int fd)
{
    //设置为非阻塞
    int ret = fcntl(fd, F_GETFL);
    if(ret < 0)
        cout << "F_GETL error: " << errno << " " << strerror(errno) << endl;
    fcntl(fd, F_SETFL, ret | O_NONBLOCK);
}

//非阻塞式IO
int main()
{
    char buffer[64] = { 0 };

    setNonBlock(0);
    LoadTask();
    
    while(true)
    {
        printf(":> ");
        fflush(stdout);
        ssize_t n = read(0, buffer, sizeof(buffer) - 1);

        if(n > 0)
        {
            buffer[n] = '\0';
            cout << "echo: " << buffer << endl;
        }
        else if(n == 0)
        {
            cout << "read done..." << endl;
            break;
        }
        else
        {
            if(errno == EAGAIN || errno == EWOULDBLOCK)
            {
                sleep(1);   //防止刷屏太快
                // cout << "Data not ready..." << endl;
                Start();
                continue;
            }
            else if(errno == EINTR)
            {
                cout << "Signal done..." << endl;
                continue;
            }
            else
            {
                cout << "read error: " << errno << " " << strerror(errno) << endl;
                break;
            }
        }
    }

    return 0;
}

// //阻塞式IO
// int main()
// {
//     char buffer[64] = { 0 };

//     ssize_t n = 1;
//     while(n)
//     {
//         printf(":> ");
//         fflush(stdout);
//         n = read(0, buffer, sizeof(buffer) - 1);

//         if(n > 0)
//         {
//             buffer[n] = '\0';
//             cout << "echo: " << buffer << endl;
//         }
//         else
//         {
//             cout << "read error: " << errno << " " << strerror(errno) << endl;
//             break;
//         }
//     }

//     return 0;
// }